package com.starry.utils;

import org.apache.commons.dbutils.QueryRunner;
import org.apache.commons.dbutils.ResultSetHandler;
import org.apache.commons.dbutils.handlers.BeanHandler;
import org.apache.commons.dbutils.handlers.BeanListHandler;
import org.apache.commons.dbutils.handlers.ScalarHandler;

import java.sql.Connection;
import java.util.List;

public class DBUtils {
    /**
     * 执行 insert/update/delete 三种sql,返回执行影响条数
     *
     * @param sql    执行sql脚本
     * @param params 自定义参数
     * @return 返回执行影响条数，-1代表执行失败
     */
    public static int ExcuteNonQuery(String sql, Object... params) {
        Connection conn = JdbcUtils.getConnection();
        QueryRunner queryRunner = new QueryRunner();
        try {
            return queryRunner.update(conn, sql, params);
        } catch (Exception e) {
            e.printStackTrace();
        } finally {
            JdbcUtils.close(conn);
        }
        return -1;
    }

    /**
     * 查询单个实体
     *
     * @param type   实体类型
     * @param sql    查询sql
     * @param params 参数
     * @param <T>    实体类型
     * @return
     */
    public static <T> T QueryModel(Class<T> type, String sql, Object... params) {
        Connection conn = JdbcUtils.getConnection();
        QueryRunner queryRunner = new QueryRunner();
        try {
            return queryRunner.query(conn, sql, new BeanHandler<T>(type), params);
        } catch (Exception e) {
            e.printStackTrace();
        } finally {
            JdbcUtils.close(conn);
        }
        return null;
    }

    /**
     * 查询实体集合
     *
     * @param type   实体类型
     * @param sql    查询sql
     * @param params 参数
     * @param <T>    实体类型
     * @return
     */
    public static <T> List<T> QueryList(Class<T> type, String sql, Object... params) {
        Connection conn = JdbcUtils.getConnection();
        QueryRunner queryRunner = new QueryRunner();
        try {
            return queryRunner.query(conn, sql, new BeanListHandler<T>(type), params);
        } catch (Exception e) {
            e.printStackTrace();
        } finally {
            JdbcUtils.close(conn);
        }
        return null;
    }

    /**
     * 查询单个值
     * @param sql    查询sql
     * @param params 参数
     * @return
     */
    public static Object QueryScalar(String sql, Object... params) {
        Connection conn = JdbcUtils.getConnection();
        QueryRunner queryRunner = new QueryRunner();
        try {
            return queryRunner.query(conn, sql, new ScalarHandler(), params);
        } catch (Exception e) {
            e.printStackTrace();
        } finally {
            JdbcUtils.close(conn);
        }
        return null;
    }
}
